<?php  if ( ! defined('BASEPATH')) exit('No direct script access allowed');

class PW_Pagination extends CI_Pagination
{
    var $page_uri_array = FALSE;
    var $segment_start  = 3;
    var $use_per_page_segment = FALSE;
    var $per_page_segment = 'per_page';
    var $cur_page = null;
    
    /**
     * Constructor
     *
     * @access public
     * @param  array    initialization parameters
     */
    public function initialize($params = null) {
        
        parent::initialize($params);
        
        if (isset($params['use_per_page_segment'])) {
            $this->_process_per_page();
        }
        
        if ($this->cur_page === null) {
            $this->_process_current_page();
        }
    }
    
    /**
     * Process current page and and number of items
     *
     * @access private
     */
    private function _process_current_page() {
        
        $CI =& get_instance();
        
        // Set the base page index for starting page number
        if ($this->use_page_numbers) {
            $base_page = 1;
        } else {
            $base_page = 0;
        }
        
        if ($CI->config->item('enable_query_strings') === TRUE OR $this->page_query_string === TRUE) {
            if ($CI->input->get($this->query_string_segment) != $base_page) {
                $this->cur_page = $CI->input->get($this->query_string_segment);
        
                // Prep the current page - no funny business!
                $this->cur_page = (int) $this->cur_page;
            }
        } else {
            if ($this->page_uri_array) {
                // Get uri array
                $vars = $CI->uri->uri_to_assoc($this->segment_start);
        
                $this->cur_page = (int) $vars[$this->query_string_segment];
            } elseif ($CI->uri->segment($this->uri_segment) != $base_page) {
                $this->cur_page = $CI->uri->segment($this->uri_segment);
        
                // Prep the current page - no funny business!
                $this->cur_page = (int) $this->cur_page;
            }
        }
        
        // Set current page to 1 if using page numbers instead of offset
        if ($this->use_page_numbers AND $this->cur_page == 0) {
            $this->cur_page = $base_page;
        }
        
        if (!is_numeric($this->cur_page)) {
            $this->cur_page = $base_page;
        }
    }
    
    private function _process_per_page() {
        
        $CI =& get_instance();
        if ($this->page_uri_array) {
            // Get uri array
            $vars = $CI->uri->uri_to_assoc($this->segment_start);
        }
        
        // get items number vie uri
        if ($this->use_per_page_segment == TRUE) {
            if (($CI->config->item('enable_query_strings') === TRUE OR $this->page_query_string === TRUE)) {
                if ($per_page = $CI->input->get($this->per_page_segment)) {
                    $this->per_page = $per_page;
                }
            }
            else {
                if (!$this->page_uri_array) {
                    if ($per_page = $CI->uri->segment($this->per_page_segment)) {
                        $this->per_page = $per_page;
                    }
                } else {
                    // Get uri array
                    if (isset($vars[$this->per_page_segment])) {
                        $this->per_page = $vars[$this->per_page_segment];
                    }
                }
            }
        }
    }
    
    public function create_links() {
        
        // If our item count or per-page total is zero there is no need to continue.
        if ($this->total_rows == 0 OR $this->per_page == 0) {
            return '';
        }

        // Calculate the total number of pages
        $num_pages = ceil($this->total_rows / $this->per_page);

        // Is there only one page? Hm... nothing more to do here then.
        if ($num_pages == 1) {
            return '';
        }

        // Set the base page index for starting page number
        if ($this->use_page_numbers) {
            $base_page = 1;
        } else {
            $base_page = 0;
        }

        // Determine the current page number.
        $CI =& get_instance();
        
        // If no base url, auto-load current url
        if ($this->base_url == '') {
            $CI->load->helper('url');
            
            $this->base_url = current_url();
        }
        
        if ($this->page_uri_array) {
            // Get uri array
            $vars = $CI->uri->uri_to_assoc($this->segment_start);
        }
        
        // Add per page segment string
        if ($this->use_per_page_segment == TRUE) {
            if (($CI->config->item('enable_query_strings') === TRUE) OR ($this->page_query_string === TRUE)) {
                if (!$CI->input->get($this->per_page_segment)) {
                    $this->base_url .= '&amp;' . $this->per_page_segment . '=' . $this->per_page;
                }
            }
            elseif ($this->page_uri_array) {
                if (!isset($vars[$this->per_page_segment])) {
                    $vars[$this->per_page_segment] = $this->per_page;
                }
            }
        }

        // Is the page number beyond the result range?
        // If so we show the last page
        if ($this->use_page_numbers) {
            if ($this->cur_page > $num_pages) {
                $this->cur_page = $num_pages;
            }
        } else {
            if ($this->cur_page > $this->total_rows) {
                $this->cur_page = ($num_pages - 1) * $this->per_page;
            }
        }

        $this->num_links = (int)$this->num_links;
        
        if ($this->num_links < 1) {
            show_error('Your number of links must be a positive number.');
        }
        
        $uri_page_number = $this->cur_page;
        
        if (!$this->use_page_numbers) {
            $this->cur_page = floor(($this->cur_page/$this->per_page) + 1);
        }

        // Calculate the start and end numbers. These determine
        // which number to start and end the digit links with
        $start = (($this->cur_page - $this->num_links) > 0) ? $this->cur_page - ($this->num_links - 1) : 1;
        $end   = (($this->cur_page + $this->num_links) < $num_pages) ? $this->cur_page + $this->num_links : $num_pages;
        
        // Is pagination being used over GET or POST?  If get, add a per_page query
        // string. If post, add a trailing slash to the base URL if needed
        if ($CI->config->item('enable_query_strings') === TRUE OR $this->page_query_string === TRUE) {
            $this->base_url = rtrim($this->base_url).'&amp;'.$this->query_string_segment.'=';
        } else {
            if ($this->page_uri_array == TRUE) {
                if ($segment_start = $CI->uri->segment($this->segment_start)) {
                    $this->base_url = rtrim(substr($this->base_url, 0, strpos($this->base_url, $segment_start)), '/') . '/';
                }
            }
            $this->base_url = rtrim($this->base_url, '/') .'/';
        }
        // And here we go...
        $output = '';
        
        // Render the "First" link
        if  ($this->first_link !== FALSE AND $this->cur_page > ($this->num_links + 1)) {
            if (!$this->first_url == '') {
                $first_url = $this->first_url;
            } else {
                $first_url = $this->base_url;
                if ($this->page_uri_array) {
                    $vars[$this->query_string_segment] = $base_page;
                    $first_url .= $CI->uri->assoc_to_uri($vars);
                }
            }
            
            $output .= $this->first_tag_open.'<a '.$this->anchor_class.'href="'.$first_url.'">'.$this->first_link.'</a>'.$this->first_tag_close;
        }

        // Render the "previous" link
        if  ($this->prev_link !== FALSE AND $this->cur_page != 1) {
            if ($this->use_page_numbers) {
                $i = $uri_page_number - 1;
            } else {
                $i = $uri_page_number - $this->per_page;
            }

            if ($i == 0 && $this->first_url != '') {
                $output .= $this->prev_tag_open.'<a '.$this->anchor_class.'href="'.$this->first_url.'">'.$this->prev_link.'</a>'.$this->prev_tag_close;
            } else {
                if (!$this->page_uri_array) {
                    $i = ($i == 0) ? '' : $this->prefix.$i.$this->suffix;
                } else {
                    $vars[$this->query_string_segment] = ($i == '') ? $base_page : $i;
                    $i = $CI->uri->assoc_to_uri($vars);
                }
                
                $output .= $this->prev_tag_open.'<a '.$this->anchor_class.'href="'.$this->base_url.$i.'">'.$this->prev_link.'</a>'.$this->prev_tag_close;
            }
        }

        // Render the pages
        if ($this->display_pages !== FALSE) {
            // Write the digit links
            for ($loop = $start -1; $loop <= $end; $loop++) {
                if ($this->use_page_numbers) {
                    $i = $loop;
                } else {
                    $i = ($loop * $this->per_page) - $this->per_page;
                }

                if ($i >= $base_page) {
                    if ($this->cur_page == $loop) {
                        $output .= $this->cur_tag_open.$loop.$this->cur_tag_close; // Current page
                    } else {
                        $n = ($i == $base_page) ? '' : $i;

                        if ($n == '' && $this->first_url != '') {
                            $output .= $this->num_tag_open.'<a '.$this->anchor_class.'href="'.$this->first_url.'">'.$loop.'</a>'.$this->num_tag_close;
                        } else {
                            if (!$this->page_uri_array) {
                                $n = ($n == '') ? '' : $this->prefix.$n.$this->suffix;
                            } else {
                                $vars[$this->query_string_segment] = ($n == '') ? $base_page : $n;
                                $n = $CI->uri->assoc_to_uri($vars);
                            }
                            
                            $output .= $this->num_tag_open.'<a '.$this->anchor_class.'href="'.$this->base_url.$n.'">'.$loop.'</a>'.$this->num_tag_close;
                        }
                    }
                }
            }
        }

        // Render the "next" link
        if ($this->next_link !== FALSE AND $this->cur_page < $num_pages) {
            if ($this->use_page_numbers) {
                $i = $this->cur_page + 1;
            } else {
                $i = ($this->cur_page * $this->per_page);
            }
            
            if (!$this->page_uri_array) {
                $i = $this->prefix.$i.$this->suffix;
            } else {
                $vars[$this->query_string_segment] = $i;
                $i = $CI->uri->assoc_to_uri($vars);
            }
            
            $output .= $this->next_tag_open.'<a '.$this->anchor_class.'href="'.$this->base_url.$i.'">'.$this->next_link.'</a>'.$this->next_tag_close;
        }

        // Render the "Last" link
        if ($this->last_link !== FALSE AND ($this->cur_page + $this->num_links) < $num_pages) {
            if ($this->use_page_numbers) {
                $i = $num_pages;
            } else {
                $i = (($num_pages * $this->per_page) - $this->per_page);
            }
            
            if (!$this->page_uri_array) {
                $i = $this->prefix.$i.$this->suffix;
            } else {
                $vars[$this->query_string_segment] = $i;
                $i = $CI->uri->assoc_to_uri($vars);
            }
            
            $output .= $this->last_tag_open.'<a '.$this->anchor_class.'href="'.$this->base_url.$i.'">'.$this->last_link.'</a>'.$this->last_tag_close;
        }

        // Kill double slashes.  Note: Sometimes we can end up with a double slash
        // in the penultimate link so we'll kill all double slashes.
        $output = preg_replace("#([^:])//+#", "\\1/", $output);

        // Add the wrapper HTML if exists
        $output = $this->full_tag_open.$output.$this->full_tag_close;

        return $output;
    }
}

?>